from config import to_abs_path


class BertConfig:
    def __init__(self):
        self.pretrained_model_path = to_abs_path('model/bert/bert-base-chinese')
        self.log_batch = 5
        self.epochs = 10
        self.lr = 5e-2
        self.model_path = to_abs_path('model/bert/bert_model.bin')
        self.onnx_model_path = to_abs_path('model/bert/onnx_bert_model.onnx')
        self.q_model_path = to_abs_path('model/bert/q_bert_model.bin')
        self.distill_hard_model_path = to_abs_path('model/bert/distill_hard_model.bin')
        self.distill_soft_model_path = to_abs_path('model/bert/distill_soft_model.bin')
        self.distill_middle_model_path = to_abs_path('model/bert/distill_middle_model.bin')
        self.distill_hard_f1_img_path = to_abs_path('img/bert/distill_hard_f1_img.png')
        self.distill_soft_f1_img_path = to_abs_path('img/bert/distill_soft_f1_img.png')
        self.distill_middle_f1_img_path = to_abs_path('img/bert/distill_middle_f1_img.png')
        self.temp_path = to_abs_path('data/bert/temp')
        self.f1_img_path = to_abs_path('img/bert_f1.png')
        self.batch_size = 1024
        self.max_length = 512


b_conf = BertConfig()
